#include <iostream>
#include <memory>
#include <functional>
#include "UdpServer.hpp"
#include "MessageRoute.hpp"

void Usage(char *proc)
{
    std::cout << "Usage:\n\t" << *proc << " local_port\n" << std::endl;
}

// using func_t = std::function<void(int sockfd, const std::string message, const InetAddr who)>;

int main(int argc, char *argv[])
{
    if (argc != 2)
    {
        Usage(argv[0]);
        exit(USAGE_ERROR);
    }

    EnableScreen();
    uint16_t port = std::stoi(argv[1]);

    // 定义消息转发模块
    MessageRoute route;

    std::unique_ptr<UdpServer> usvr=std::make_unique<UdpServer>(port,
        std::bind(&MessageRoute::Route, &route, std::placeholders::_1, std::placeholders::_2, std::placeholders::_3)
    );

    usvr->InitServer();
    usvr->Start();

    return 0;
}